op 	 {
  name: "TPUReshardVariables"
  input_arg {
    name: "vars"
    type: DT_RESOURCE
    number_attr: "N"
  }
  input_arg {
    name: "new_format_key"
    type: DT_STRING
  }
  input_arg {
    name: "format_state_var"
    type: DT_RESOURCE
  }
  attr {
    name: "N"
    type: "int"
    has_minimum: true
  }
  is_stateful: true
}
